package com.duoduo.剑指Offer;

import com.duoduo.TreeNode;

/**
 * @program: algorithm
 * @description: 平衡二叉树38
 * 输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。
 *
 *  
 *
 * 示例 1:
 *
 * 给定二叉树 [3,9,20,null,null,15,7]
 *
 *     3
 *    / \
 *   9  20
 *     /  \
 *    15   7
 * 返回 true 。
 *
 * @author: chaidl  ！
 * @create: 2022-05-28 20:05
 */
public class 平衡二叉树38 {
	//后序遍历
	public boolean isBalanced(TreeNode root) {
		return height(root) >= 0;
	}
	
	public int height(TreeNode root) {
		if (root == null) {
			return 0;
		}
		int leftHeight = height(root.left);
		int rightHeight = height(root.right);
		if (leftHeight == -1 || rightHeight == -1 || Math.abs(leftHeight - rightHeight) > 1) {
			return -1;
		} else {
			return Math.max(leftHeight, rightHeight) + 1;
		}
	}
	//先序遍历
	public boolean isBalanced2(TreeNode root) {
	//	此题解法就是递归
		if (root == null){
			return true;
		}else {
			//判断左子树和右子树的差会不会超过1，超过就不是平衡树了
			//还要判断子节点的左子树和右子树是不是平衡树
			return Math.abs(dfs(root.left)-dfs(root.right))<=1 && isBalanced2(root.left) && isBalanced2(root.right);
		}
	}
	
	private int dfs(TreeNode root) {
		//如果根节点为空就是0
		if (root == null) {
		    return 0;
		}
		//计算出此节点的树最深高度
		return Math.max(dfs(root.left),dfs(root.right))+1;
	}
}
